home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Demo / stdwin / jukebox.py < prev    next >
Encoding:
Python Source  |  2000-06-23  |  9.1 KB  |  414 lines
  1. #! /usr/bin/env python
  2.  
  3. # XXX This only works on SGIs running IRIX 4.0 or higher
  4.  
  5. # JUKEBOX: browse directories full of sampled sound files.
  6. #
  7. # One or more "list windows" display the files and subdirectories of
  8. # the arguments.  Double-clicking on a subdirectory opens a new window
  9. # displaying its contents (and so on recursively).  Double clicking
  10. # on a file plays it as a sound file (assuming it is one).
  11. #
  12. # Playing is asynchronous: the application keeps listening for events
  13. # while the sample is playing, so you can cancel playing or start a
  14. # new sample right away.  Synchronous playing is available through the
  15. # -s option.
  16. #
  17. # The control window displays a "stop button" that cancel the current
  18. # play request.
  19. #
  20. # Most sound file formats recognized by SOX or SFPLAY are recognized.
  21. # Since conversion is costly, converted files are cached in
  22. # /usr/tmp/@j* until the user quits or changes the sampling rate via
  23. # the Rate menu.
  24.  
  25. import commands
  26. import getopt
  27. import os
  28. from stat import *
  29. import rand
  30. import stdwin
  31. from stdwinevents import *
  32. import sys
  33. import tempfile
  34. import sndhdr
  35.  
  36. from WindowParent import WindowParent
  37. from Buttons import PushButton
  38.  
  39. # Pathnames
  40.  
  41. DEF_DB = '/usr/local/sounds'        # Default directory of sounds
  42. SOX = '/usr/local/bin/sox'        # Sound format conversion program
  43. SFPLAY = '/usr/sbin/sfplay'        # Sound playing program
  44.  
  45.  
  46. # Global variables
  47.  
  48. class struct: pass        # Class to define featureless structures
  49.  
  50. G = struct()            # Holds writable global variables
  51.  
  52.  
  53. # Main program
  54.  
  55. def main():
  56.     G.synchronous = 0    # If set, use synchronous audio.write()
  57.     G.debug = 0        # If set, print debug messages
  58.     G.busy = 0        # Set while asynchronous playing is active
  59.     G.windows = []        # List of open windows, except control
  60.     G.mode = ''        # File type (default any that sfplay knows)
  61.     G.rate = 0        # Sampling rate (default " " " ")
  62.     G.tempprefix = tempfile.mktemp()
  63.     #
  64.     try:
  65.         optlist, args = getopt.getopt(sys.argv[1:], 'dr:st:')
  66.     except getopt.error, msg:
  67.         sys.stdout = sys.stderr
  68.         print msg
  69.         print 'usage: jukebox [-d] [-s] [-t type] [-r rate]'
  70.         print '  -d        debugging (-dd event debugging)'
  71.         print '  -s        synchronous playing'
  72.         print '  -t type   file type'
  73.         print '  -r rate   sampling rate'
  74.         sys.exit(2)
  75.     #
  76.     for optname, optarg in optlist:
  77.         if   optname == '-d':
  78.             G.debug = G.debug + 1
  79.         elif optname == '-r':
  80.             G.rate = int(eval(optarg))
  81.         elif optname == '-s':
  82.             G.synchronous = 1
  83.         elif optname == '-t':
  84.             G.mode = optarg
  85.     #
  86.     if G.debug:
  87.         for name in G.__dict__.keys():
  88.             print 'G.' + name, '=', `G.__dict__[name]`
  89.     #
  90.     if not args:
  91.         args = [DEF_DB]
  92.     #
  93.     G.cw = opencontrolwindow()
  94.     for dirname in args:
  95.         G.windows.append(openlistwindow(dirname))
  96.     #
  97.     #
  98.     try:
  99.         maineventloop()
  100.     finally:
  101.         clearcache()
  102.         killchild()
  103.  
  104. # Entries in Rate menu:
  105. rates = ['default', '7350', \
  106.     '8000', '11025', '16000', '22050', '32000', '41000', '48000']
  107.  
  108. def maineventloop():
  109.     mouse_events = WE_MOUSE_DOWN, WE_MOUSE_MOVE, WE_MOUSE_UP
  110.     while G.windows:
  111.         try:
  112.             type, w, detail = event = stdwin.getevent()
  113.         except KeyboardInterrupt:
  114.             killchild()
  115.             continue
  116.         if w == G.cw.win:
  117.             if type == WE_CLOSE:
  118.                 return
  119.             if type == WE_TIMER:
  120.                 checkchild()
  121.                 if G.busy:
  122.                     G.cw.win.settimer(1)
  123.             elif type == WE_MENU:
  124.                 menu, item = detail
  125.                 if menu is G.ratemenu:
  126.                     clearcache()
  127.                     if item == 0:
  128.                         G.rate = 0
  129.                     else:
  130.                         G.rate = eval(rates[item])
  131.                     for i in range(len(rates)):
  132.                         menu.check(i, (i == item))
  133.             else:
  134.                 G.cw.dispatch(event)
  135.         else:
  136.             if type == WE_DRAW:
  137.                 w.drawproc(w, detail)
  138.             elif type in mouse_events:
  139.                 w.mouse(w, type, detail)
  140.             elif type == WE_CLOSE:
  141.                 w.close(w)
  142.                 del w, event
  143.             else:
  144.                 if G.debug > 1: print type, w, detail
  145.  
  146. def checkchild():
  147.     if G.busy:
  148.         waitchild(1)
  149.  
  150. def killchild():
  151.     if G.busy:
  152.         os.kill(G.busy, 9)
  153.         waitchild(0)
  154.  
  155. def waitchild(options):
  156.     pid, sts = os.waitpid(G.busy, options)
  157.     if pid == G.busy:
  158.         G.busy = 0
  159.         G.stop.enable(0)
  160.  
  161.  
  162. # Control window -- to set gain and cancel play operations in progress
  163.  
  164. def opencontrolwindow():
  165.     stdwin.setdefscrollbars(0, 0)
  166.     cw = WindowParent().create('Jukebox', (0, 0))
  167.     #
  168.     stop = PushButton().definetext(cw, '        Stop        ')
  169.     stop.hook = stop_hook
  170.     stop.enable(0)
  171.     G.stop = stop
  172.     #
  173.     cw.realize()
  174.     #
  175.     G.ratemenu = cw.win.menucreate('Rate')
  176.     for r in rates:
  177.         G.ratemenu.additem(r)
  178.     if G.rate == 0:
  179.         G.ratemenu.check(0, 1)
  180.     else:
  181.         for i in len(range(rates)):
  182.             if rates[i] == `G.rate`:
  183.                 G.ratemenu.check(i, 1)
  184.     #
  185.     return cw
  186.  
  187. def stop_hook(self):
  188.     killchild()
  189.  
  190.  
  191. # List windows -- to display list of files and subdirectories
  192.  
  193. def openlistwindow(dirname):
  194.     list = os.listdir(dirname)
  195.     list.sort()
  196.     i = 0
  197.     while i < len(list):
  198.         if list[i][0] == '.':
  199.             del list[i]
  200.         else:
  201.             i = i+1
  202.     for i in range(len(list)):
  203.         fullname = os.path.join(dirname, list[i])
  204.         if os.path.isdir(fullname):
  205.             info = '/'
  206.         else:
  207.             try:
  208.                 size = os.stat(fullname)[ST_SIZE]
  209.                 info = `(size + 1023)/1024` + 'k'
  210.             except IOError:
  211.                 info = '???'
  212.             info = '(' + info + ')'
  213.         list[i] = list[i], info
  214.     width = maxwidth(list)
  215.     # width = width + stdwin.textwidth(' ')    # XXX X11 stdwin bug workaround
  216.     height = len(list) * stdwin.lineheight()
  217.     stdwin.setdefwinsize(width, min(height, 500))
  218.     stdwin.setdefscrollbars(0, 1)
  219.     w = stdwin.open(dirname)
  220.     stdwin.setdefwinsize(0, 0)
  221.     w.setdocsize(width, height)
  222.     w.drawproc = drawlistwindow
  223.     w.mouse = mouselistwindow
  224.     w.close = closelistwindow
  225.     w.dirname = dirname
  226.     w.list = list
  227.     w.selected = -1
  228.     return w
  229.  
  230. def maxwidth(list):
  231.     width = 1
  232.     for name, info in list:
  233.         w = stdwin.textwidth(name + '  ' + info)
  234.         if w > width: width = w
  235.     return width
  236.  
  237. def drawlistwindow(w, area):
  238. ##    (left, top), (right, bottom) = area
  239.     d = w.begindrawing()
  240.     d.erase((0, 0), (1000, 10000))
  241.     lh = d.lineheight()
  242.     h, v = 0, 0
  243.     for name, info in w.list:
  244.         if info == '/':
  245.             text = name + '/'
  246.         else:
  247.             text = name + '  ' + info
  248.         d.text((h, v), text)
  249.         v = v + lh
  250.     showselection(w, d)
  251.     d.close()
  252.  
  253. def hideselection(w, d):
  254.     if w.selected >= 0:
  255.         invertselection(w, d)
  256.  
  257. def showselection(w, d):
  258.     if w.selected >= 0:
  259.         invertselection(w, d)
  260.  
  261. def invertselection(w, d):
  262.     lh = d.lineheight()
  263.     h1, v1 = p1 = 0, w.selected*lh
  264.     h2, v2 = p2 = 1000, v1 + lh
  265.     d.invert(p1, p2)
  266.  
  267. def mouselistwindow(w, type, detail):
  268.     (h, v), clicks, button = detail[:3]
  269.     d = w.begindrawing()
  270.     lh = d.lineheight()
  271.     if 0 <= v < lh*len(w.list):
  272.         i = v / lh
  273.     else:
  274.         i = -1
  275.     if w.selected <> i:
  276.         hideselection(w, d)
  277.         w.selected = i
  278.         showselection(w, d)
  279.     d.close()
  280.     if type == WE_MOUSE_DOWN and clicks >= 2 and i >= 0:
  281.         setcursors('watch')
  282.         name, info = w.list[i]
  283.         fullname = os.path.join(w.dirname, name)
  284.         if info == '/':
  285.             if clicks == 2:
  286.                 G.windows.append(openlistwindow(fullname))
  287.         else:
  288.             playfile(fullname)
  289.         setcursors('cross')
  290.  
  291. def closelistwindow(w):
  292.     G.windows.remove(w)
  293.  
  294. def setcursors(cursor):
  295.     for w in G.windows:
  296.         w.setwincursor(cursor)
  297.     G.cw.win.setwincursor(cursor)
  298.  
  299.  
  300. # Playing tools
  301.  
  302. cache = {}
  303.  
  304. def clearcache():
  305.     for x in cache.keys():
  306.         cmd = 'rm -f ' + cache[x]
  307.         if G.debug: print cmd
  308.         sts = os.system(cmd)
  309.         if sts:
  310.             print cmd
  311.             print 'Exit status', sts
  312.         del cache[x]
  313.  
  314. validrates = (8000, 11025, 16000, 22050, 32000, 44100, 48000)
  315.  
  316. def playfile(filename):
  317.     killchild()
  318.     try:
  319.         tuple = sndhdr.what(filename)
  320.     except IOError, msg:
  321.         print 'Can\'t open', filename, msg
  322.         stdwin.fleep()
  323.         return
  324.     raw = 0
  325.     if tuple:
  326.         mode, rate = tuple[:2]
  327.         if rate == 0:
  328.             rate = G.rate
  329.             if rate == 0:
  330.                 rate = 8000
  331.     else:
  332.         mode = G.mode
  333.         rate = G.rate
  334.     if G.debug: print 'mode =', mode, 'rate =', rate
  335.     if mode in ('au', 'aiff', 'wav', 'aifc', 'ul', 'ub', 'sb') and \
  336.           rate in validrates:
  337.         tempname = filename
  338.         if mode in ('ul', 'ub', 'sb'):
  339.             raw = 1
  340.     elif cache.has_key(filename):
  341.         tempname = cache[filename]
  342.     else:
  343.         tempname = G.tempprefix + `rand.rand()` + '.aiff'
  344.         cmd = SOX
  345.         if G.debug:
  346.             cmd = cmd + ' -V'
  347.         if mode <> '':
  348.             cmd = cmd + ' -t ' + mode
  349.         cmd = cmd + ' ' + commands.mkarg(filename)
  350.         cmd = cmd + ' -t aiff'
  351.         if rate not in validrates:
  352.             rate = 32000
  353.         if rate:
  354.             cmd = cmd + ' -r ' + `rate`
  355.         cmd = cmd + ' ' + tempname
  356.         if G.debug: print cmd
  357.         sts = os.system(cmd)
  358.         if sts:
  359.             print cmd
  360.             print 'Exit status', sts
  361.             stdwin.fleep()
  362.             try:
  363.                 os.unlink(tempname)
  364.             except:
  365.                 pass
  366.             return
  367.         cache[filename] = tempname
  368.     if raw:
  369.         pid = sfplayraw(tempname, tuple)
  370.     else:
  371.         pid = sfplay(tempname, [])
  372.     if G.synchronous:
  373.         sts = os.wait(pid, 0)
  374.     else:
  375.         G.busy = pid
  376.         G.stop.enable(1)
  377.         G.cw.win.settimer(1)
  378.  
  379. def sfplayraw(filename, tuple):
  380.     args = ['-i']
  381.     type, rate, channels, frames, bits = tuple
  382.     if type == 'ul':
  383.         args.append('mulaw')
  384.     elif type == 'ub':
  385.         args = args + ['integer', '8', 'unsigned']
  386.     elif type == 'sb':
  387.         args = args + ['integer', '8', '2scomp']
  388.     else:
  389.         print 'sfplayraw: warning: unknown type in', tuple
  390.     if channels > 1:
  391.         args = args + ['channels', `channels`]
  392.     if not rate:
  393.         rate = G.rate
  394.     if rate:
  395.         args = args + ['rate', `rate`]
  396.     args.append('end')
  397.     return sfplay(filename, args)
  398.  
  399. def sfplay(filename, args):
  400.     if G.debug:
  401.         args = ['-p'] + args
  402.     args = [SFPLAY, '-r'] + args + [filename]
  403.     if G.debug: print 'sfplay:', args
  404.     pid = os.fork()
  405.     if pid == 0:
  406.         # Child
  407.         os.execv(SFPLAY, args)
  408.         # NOTREACHED
  409.     else:
  410.         # Parent
  411.         return pid
  412.  
  413. main()
  414.